package com.lagou.edu.factory;

import com.alibaba.druid.util.StringUtils;
import com.czp.annotation.MyAutowired;
import com.czp.annotation.MyService;
import com.czp.annotation.MyTransactional;
import com.lagou.edu.utils.ClassUtil;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import java.io.InputStream;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.*;

/**
 * @author 应癫
 * <p>
 * 工厂类，生产对象（使用反射技术）
 */
public class BeanFactory {

    /**
     * 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
     * 任务二：对外提供获取实例对象的接口（根据id获取）
     */

    private static Map<String, Object> map = new HashMap<>();  // 存储对象
    private static String packageName;

    static {
//        parseXmlBeans("beans.xml");
        parseAnnotation("beans.xml");
    }
    private static void parseXmlBeans(String classpath) {
        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
        // 加载xml
        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream(classpath);
        // 解析xml
        SAXReader saxReader = new SAXReader();
        try {
            Document document = saxReader.read(resourceAsStream);
            Element rootElement = document.getRootElement();
            List<Element> beanList = rootElement.selectNodes("//bean");
            for (int i = 0; i < beanList.size(); i++) {
                Element element = beanList.get(i);
                // 处理每个bean元素，获取到该元素的id 和 class 属性
                String id = element.attributeValue("id");        // accountDao
                String clazz = element.attributeValue("class");  // com.lagou.edu.dao.impl.JdbcAccountDaoImpl
                // 通过反射技术实例化对象
                Class<?> aClass = Class.forName(clazz);
                Object o = aClass.getDeclaredConstructor().newInstance();  // 实例化之后的对象

                // 存储到map中待用
                map.put(id, o);

            }


            // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
            // 有property子元素的bean就有传值需求
            List<Element> propertyList = rootElement.selectNodes("//property");
            // 解析property，获取父元素
            for (int i = 0; i < propertyList.size(); i++) {
                Element element = propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
                String name = element.attributeValue("name");
                String ref = element.attributeValue("ref");

                // 找到当前需要被处理依赖关系的bean
                Element parent = element.getParent();

                // 调用父元素对象的反射功能
                String parentId = parent.attributeValue("id");
                Object parentObject = map.get(parentId);
                // 遍历父对象中的所有方法，找到"set" + name
                Method[] methods = parentObject.getClass().getMethods();
                for (int j = 0; j < methods.length; j++) {
                    Method method = methods[j];
                    if (method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
                        method.invoke(parentObject, map.get(ref));
                    }
                }
                // 把处理之后的parentObject重新放到map中
                map.put(parentId, parentObject);
            }
        } catch (DocumentException | ClassNotFoundException | IllegalAccessException | InstantiationException | InvocationTargetException | NoSuchMethodException e) {
            e.printStackTrace();
        }
    }

    private static void parseAnnotation(String classpath) {
        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream(classpath);
        // 解析xml
        SAXReader saxReader = new SAXReader();
        Document document = null;
        try {
            document = saxReader.read(resourceAsStream);
        } catch (DocumentException e) {
            e.printStackTrace();
        }
        Element rootElement = document.getRootElement();
        List<Element> beanList = rootElement.selectNodes("//component-scan");
        packageName = beanList.get(0).attributeValue("base-package");

        /**
         * 扫包
         */
        List<Class<?>> classes = ClassUtil.getClasses(packageName);
        /**
         * 初始化beans
         */
        initAnnotationBeans(classes);
        /**
         * 解析 autowired注解 自动装配
         */
        parseAutowired();

        /**
         * 初始化标记了 事务注解的beans
         */
        initTransaction(classes);


    }

    private static void parseAutowired() {

        for (Map.Entry<String,Object> entry : map.entrySet()) {
            // 依赖注入
            doParseAutowired(entry.getValue());
        }
    }

    private static void doParseAutowired(Object o) {
        // 1.使用反射机制获取当前类的所有属性
        Field[] declaredFields = o.getClass().getDeclaredFields();
        // 2.判断当前类是否存在注解
        for (Field field : declaredFields) {
            MyAutowired annotation = field.getAnnotation(MyAutowired.class);
            if (annotation == null) {
                continue;
            }

            // 获取属性名称
            String name = field.getName();

            System.out.println(field.getType());
            System.out.println(ServiceLoader.load(field.getType()));
            if(field.getType().isInterface()){
                ServiceLoader<?> load = ServiceLoader.load(field.getType());
                for (Object aClass : load) {
                    System.out.println(aClass);
                }
            }

            // 根据beanName查找对象
            Object newBean = getBean(name);
            // 3.默认使用属性名称,查找bean容器对象
            field.setAccessible(true);
            // 给属性赋值 将对象注入到 属性中
            try {
                field.set(o, newBean);
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
            System.out.println("依赖注入:" + o + ";" + newBean);
        }
    }

    private static void initAnnotationBeans(List<Class<?>> classes) {
        // 2.判断是否有注解
        doInitAnnotationBeans(classes);

    }

    private static void doInitAnnotationBeans(List<Class<?>> classes) {
        for (Class<?> class1 : classes) {
            MyService annotation = class1.getAnnotation(MyService.class);
            if (annotation == null) {
                continue;
            }
            if (class1.isInterface()){
                throw new RuntimeException("");
            }
            // beanId 类名小写
            String beanId = annotation.value();
            if (StringUtils.isEmpty(beanId)) {
                // 获取当前类名
                beanId = Character.toLowerCase(class1.getSimpleName().charAt(0)) + class1.getSimpleName().substring(1);
            }
            Object newInstance = null;
            try {
                newInstance = class1.getDeclaredConstructor().newInstance();
            } catch (InstantiationException | IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
                e.printStackTrace();
            }
            map.put(beanId, newInstance);
        }
    }

    private static void initTransaction(List<Class<?>> classes) {
        doInitTransaction(classes);
    }

    private static void doInitTransaction(List<Class<?>> classes) {
        for (Class<?> class1 : classes) {
            MyService annotation = class1.getAnnotation(MyService.class);
            if (annotation == null) {
                continue;
            }
            MyTransactional tranAnnotation = class1.getAnnotation(MyTransactional.class);
            MyTransactional methodAnnotation = null;
            /**
             * 获取所有方法 如果其中一个方法或者整个类都注解了事务注解,  则将bean转成动态代理对象
             */
            Method[] declaredMethods = class1.getDeclaredMethods();
            for(Method method : declaredMethods){
                methodAnnotation = method.getAnnotation(MyTransactional.class);
                /**
                 * 已经有一个方法有事务注解
                 */
                if(methodAnnotation != null){
                    break;
                }
            }
            if (tranAnnotation == null && methodAnnotation == null) {
                continue;
            }
            // beanId 类名小写
            String beanId = annotation.value();

            ProxyFactory proxyFactory = (ProxyFactory) getBean("proxyFactory");
            /**
             * 实现了接口
             */
            Class<?>[] interfaces = class1.getInterfaces();
            if (interfaces.length > 0) {
                map.put(beanId, proxyFactory.getJdkProxy(map.get(beanId)));
            } else {
                /**
                 * 未实现接口
                 */
                map.put(beanId, proxyFactory.getCglibProxy(map.get(beanId)));
            }

        }
    }


    // 任务二：对外提供获取实例对象的接口（根据id获取）
    public static Object getBean(String id) {
        return map.get(id);
    }

}
